package InternetCode.Socket.Exa11;


import java.io.*;
import java.net.*;
import java.util.Scanner;


/**
 * 利用Socket发送和接收图片，音频，视频等数据文件
 * 通过使用DataInoutStream/DataOutputStream即可
 */
public class ServerSocketFrame{

    private File imgFile = null;// 声明所选择图片的File对象
    private DataOutputStream out = null; // 创建流对象
    private DataInputStream in = null; // 创建流对象
    private ServerSocket server; // 声明ServerSocket对象
    private Socket socket; // 声明Socket对象socket
    private long lengths = -1; // 图片文件的大小
    private Scanner scanner;

    /**
     * 新建服务器对象并等待连接
     */
    public void getServer() {
        try {
            server = new ServerSocket(1978); // 实例化Socket对象
            while (true) { // 如果套接字是连接状态
                socket = server.accept(); // 实例化Socket对象
                System.out.println("连接成功！");
                out = new DataOutputStream(socket.getOutputStream());// 获得输出流对象
                in = new DataInputStream(socket.getInputStream());// 获得输入流对象
                getClientInfo(); // 调用getClientInfo()方法
            }
        } catch (Exception e) {
            e.printStackTrace(); // 输出异常信息
        }
    }

    /**
     * 等待客户端发送数据并接收
     */
    private void getClientInfo() {
        try {
            while (true){
                // 读取数据文件大小
                long lengths = in.readLong();
                if(lengths>0){
                    String fileName=in.readUTF();
                    // 创建字节数组
                    byte[] bt = new byte[(int) lengths];
                    for (int i = 0; i < bt.length; i++) {
                        bt[i] = in.readByte();// 读取字节信息并存储到字节数组
                    }
                    File img=new File(String.valueOf(Math.random())+fileName);
                    OutputStream out=new DataOutputStream(new FileOutputStream(img));
                    out.write(bt);
                    System.out.println("文件接收成功！！");
                }
            }
        } catch (Exception e) {
        } finally {
            try {
                if (in != null) {
                    in.close();// 关闭流
                }
                if (socket != null) {
                    socket.close(); // 关闭套接字
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 发送数据文件给客户端
     */
    private void sendData(){
        while (true){
            scanner=new Scanner(System.in);
            System.out.println("请输入要发送文件全路径:");
            String path=scanner.nextLine();
            imgFile=new File(path);
            try {
                DataInputStream inStream = null;// 定义数据输入流对象
                if (imgFile != null) {
                    lengths = imgFile.length();// 获得选择图片的大小
                    inStream = new DataInputStream(new FileInputStream(imgFile));// 创建输入流对象
                } else {
                    System.out.println("没有选择文件。");
                    return;
                }
                out.writeLong(lengths);// 将文件的大小写入输出流
                out.writeUTF(imgFile.getName());
                byte[] bt = new byte[(int) lengths];// 创建字节数组
                int len = -1;
                while ((len = inStream.read(bt)) != -1) {// 将图片文件读取到字节数组
                    out.write(bt);// 将字节数组写入输出流
                }
            } catch (IOException e1) {
                e1.printStackTrace();
            }
        }
    }

    public static void main(String[] args) { // 主方法
        ServerSocketFrame frame = new ServerSocketFrame(); // 创建本类对象

        //此线程用于监听接收消息
        new Thread(new Runnable() {
            @Override
            public void run() {
                frame.getServer(); // 调用方法
            }
        }).start();
        //此线程用发送消息给服务器
        new Thread(new Runnable() {
            @Override
            public void run() {
                frame.sendData(); // 调用连接方法
            }
        }).start();
    }

}
